#pragma once

#include <torch/torch.h>
#include <opencv2/core/core.hpp>
#include <opencv2/imgcodecs.hpp>
#include <opencv2/imgproc.hpp>

cv::Mat imreadRGB(const std::string& vFilename);
void imwriteRGB(const std::string& vFilename, const cv::Mat& vImage);
cv::Mat floatNxNtensorToMat(const torch::Tensor& vTensor);
torch::Tensor floatNxNMatToTensor(const cv::Mat& vMat);
cv::Mat tensorToImage(const torch::Tensor& vTensor);
torch::Tensor imageToTensor(const cv::Mat& vImage);